numpy == 1.21.4
pandas == 1.3.5
scipy == 1.7.3
scikit-learn == 1.0.1
torch == 1.4.0
torchvision == 0.5.0 -f https://download.pytorch.org/whl/cu100/torch_stable.html
matplotlib == 3.5.1
